/*
 *    Copyright © OpenAtom Foundation.
 *
 *    Licensed under the Apache License, Version 2.0 (the "License");
 *    you may not use this file except in compliance with the License.
 *    You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0
 *
 *    Unless required by applicable law or agreed to in writing, software
 *    distributed under the License is distributed on an "AS IS" BASIS,
 *    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *    See the License for the specific language governing permissions and
 *    limitations under the License.
 */

package com.inspur.edp.bff.spi.action.query;

import com.inspur.edp.bef.api.lcp.IStandardLcp;
import com.inspur.edp.bff.api.exception.BffFrameworkException;
import com.inspur.edp.bff.api.exception.ErrorCodes;
import com.inspur.edp.bff.api.manager.SourceConfig;
import com.inspur.edp.bff.api.manager.SourceType;
import com.inspur.edp.bff.api.manager.context.QueryContext;
import com.inspur.edp.bff.spi.utils.AuthUtils;
import com.inspur.edp.bff.spi.utils.BffActionNames;
import com.inspur.edp.cef.entity.entity.IEntityData;

import java.util.ArrayList;
import java.util.HashMap;

public class DefaultQueryAction extends AbstractQueryAction {


    public DefaultQueryAction(QueryContext context) {
        super(context);
    }

    @Override
    protected void execute() {
        ArrayList<SourceConfig> sourceConfigs = getContext().getSourceConfigs();
        if (sourceConfigs == null || sourceConfigs.size() == 0) {
            throw new BffFrameworkException(ErrorCodes.BFF_FRAMEWORK_2032);
        }
        if (getContext().getSourceConfigs() == null || getContext().getSourceConfigs().get(0).getSourceType() == SourceType.Be) {
            IStandardLcp lcp = getLcp();
            if (getQueryContext().getUseCustomAuth() && getQueryContext().getAuthInfo() != null) {

                AuthUtils.buildAuthInfo(
                        lcp.getBefContext().getAuthInfo(),
                        getQueryContext().getAuthInfo().getExtType(),
                        getQueryContext().getAuthInfo().getExtend1(),
                        getQueryContext().getAuthInfo().getExtend2(),
                        getQueryContext().getAuthInfo().getExtend3(),
                        getQueryContext().getAuthInfo().getExtend4(),
                        getQueryContext().getAuthInfo().getExtend5()
                );
            }
            ArrayList<IEntityData> queryResult;
            //兼容之前
            if (this.getContext().getRootEntityPropertyMapper() != null) {

                String voNodeCode = getContext().getBECodeFromVM(getContext().getRootEntityPropertyMapper().getVOCode());
                queryResult = new ArrayList(lcp.query(voNodeCode, getQueryContext().getFilter()));
            } else {

                queryResult = new ArrayList(lcp.query(this.getQueryContext().getFilter()));
            }

            getQueryContext().setQueryResult(queryResult);
            if (queryResult == null) {
                setResult(null);
            }
            HashMap<String, ArrayList<IEntityData>> result = new HashMap<>();
            result.put(BffActionNames.DefaultName, queryResult);
            setResult(result);
        } else {
            //var queryResult = ServiceManager.GetService<IQOManagerFactory>().GetQueryManager(Context.SourceConfigs[0].SourceConfigId).Query(QueryContext.Param, QueryContext.PageInfo, QueryContext.ExtendCond).Data;
            //QueryContext.QueryResult = queryResult;
            //if (queryResult == null)
            //    Result = null;
            //Dictionary<string, List<IEntityData>> result = new Dictionary<string, List<IEntityData>>();
            //result.Add("Default", queryResult);
            //Result = result;
        }
    }

}
